from skorch.core import *
from skorch.layers import *
from skorch.vision import *
from skorch.callbacks.hooks import *
from skorch.model import create_cnn

UNet = models.OriginalUNet
FrrNet = models.FRRN

__all__ = ['UNet', 'FrrNet']

